from selenium import webdriver
from selenium.webdriver.common.by import By
import time


class Keywords():
    # 函数第一个参数必须是实例对象本身，self代表实例对象本身，以self开头可共所有方法使用，把self换位this同样，
    def __init__(self, url):
        self.wd = webdriver.Chrome()
        self.wd.maximize_window()
        self.wd.get(url)

    # 实现搜索
    def search(self, words):
        self.wd.find_element(By.ID, "sb_form_q").send_keys(words)
        self.wd.find_element(By.ID, "search_icon").click()

    # 屏幕截图保存
    def save(self):
        pic_time = time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime(time.time()))
        self.wd.save_screenshot(pic_time + ".png")

    # 退出浏览器
    def quit(self):
        self.wd.quit()
